#!/usr/bin/env python
# -*- coding:utf-8 -*-

# DrissionPage 库 文档地址 http://g1879.gitee.io/drissionpagedocs/

#-导入库
from DrissionPage import ChromiumPage,ChromiumOptions

#-配置类
class Config:

    url='https://baidu.com'
    port=7878


#-创建配置对象
co=ChromiumOptions()

#-启动配置
co.set_local_port(Config.port)
co.ignore_certificate_errors(True)

#-创建浏览器
page = ChromiumPage(addr_or_opts=co)
#-创建标签页
tab=page.new_tab()




js_code='''
// 定义一个函数，用于hook目标函数
function hookFunction(originalFunction) {
  // 保存原始的函数
  let original = originalFunction;

  // 返回一个新的函数来替代原始函数
  return function() {
    // 添加当前日期
    const currentDate = new Date().toISOString();
    // 在调用原始函数时添加日期前缀
    return original.call(this, `[${currentDate}] 骚神库 `, ...arguments);
  };
}

// 示例：hook console.log
console.log = hookFunction(console.log);

// 测试
console.log("Hello, world!");

'''
tab.add_init_js(js_code)
#-打开网址
tab.get(Config.url)
tab.run_js('console.log(document.URL)')



input('go on ?')
  